package com.stottlerhenke.simbionic.editor.common;

import java.awt.Toolkit;
import java.awt.datatransfer.Clipboard;
import java.awt.datatransfer.DataFlavor;
import java.awt.datatransfer.StringSelection;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.KeyEvent;

import javax.swing.JComponent;
import javax.swing.JOptionPane;
import javax.swing.JTable;
import javax.swing.KeyStroke;

/**
 * CopyAndPasteAdapter enables Copy-Paste Clipboard functionality on JTables.
 * The clipboard data format used by the adapter is compatible with
 * the clipboard format used by Excel. This provides for clipboard
 * interoperability between enabled JTables and Excel.
 */
public class CopyAndPasteAdapter implements ActionListener
   {
   private String rowstring,value;
   private Clipboard system;
   private StringSelection stsel;
   private JTable jTable1 ;
   /**
    * The Excel Adapter is constructed with a
    * JTable on which it enables Copy-Paste and acts
    * as a Clipboard listener.
    */
   public CopyAndPasteAdapter(JTable myJTable)
   {
	   jTable1 = myJTable;
	   KeyStroke copy = KeyStroke.getKeyStroke(KeyEvent.VK_C,ActionEvent.CTRL_MASK,false);
	   KeyStroke paste = KeyStroke.getKeyStroke(KeyEvent.VK_V,ActionEvent.CTRL_MASK,false);

	   jTable1.registerKeyboardAction(this,"Copy",copy,JComponent.WHEN_FOCUSED);
	   jTable1.registerKeyboardAction(this,"Paste",paste,JComponent.WHEN_FOCUSED);

	   system = Toolkit.getDefaultToolkit().getSystemClipboard();
   }
   
   /**
    * Public Accessor methods for the Table on which this adapter acts.
    */
	public JTable getJTable() {return jTable1;}
	/**
	 * Public Accessor methods for setting the table on which the adapter acts
	 * @param jTable1
	 */
	public void setJTable(JTable jTable1) {this.jTable1=jTable1;}
   
	/**
    * This method is activated on the Keystrokes we are listening to
    * in this implementation. Here it listens for Copy and Paste ActionCommands.
    * Selections comprising non-adjacent cells result in invalid selection and
    * then copy action cannot be performed.
    * Paste is done by aligning the upper left corner of the selection with the
    * 1st element in the current selection of the JTable.
    */
	public void actionPerformed(ActionEvent e)
    {
      if (e.getActionCommand().compareTo("Copy")==0)
      {
         StringBuffer sbf=new StringBuffer();
         // Check to ensure we have selected only a contiguous block of
         // cells
         int numcols=jTable1.getSelectedColumnCount();
         int numrows=jTable1.getSelectedRowCount();
         int[] rowsselected=jTable1.getSelectedRows();
         int[] colsselected=jTable1.getSelectedColumns();
         if (!((numrows-1==rowsselected[rowsselected.length-1]-rowsselected[0] &&
                numrows==rowsselected.length) &&
                (numcols-1==colsselected[colsselected.length-1]-colsselected[0] &&
                numcols==colsselected.length)))
         {
            JOptionPane.showMessageDialog(null, "Invalid Copy Selection",
                                          "Invalid Copy Selection",
                                          JOptionPane.ERROR_MESSAGE);
            return;
         }
         for (int i=0;i<numrows;i++)
         {
            for (int j=0;j<numcols;j++)
            {
            	sbf.append(jTable1.getValueAt(rowsselected[i],colsselected[j]));
            	if (j<numcols-1) sbf.append("\t");
            }
            sbf.append("\r\n"); // excel accept \r\n only
         }
         stsel  = new StringSelection(sbf.toString());
         system = Toolkit.getDefaultToolkit().getSystemClipboard();
         system.setContents(stsel,stsel);
      }
      if (e.getActionCommand().compareTo("Paste")==0)
      {
          System.out.println("Trying to Paste");
          int startRow=(jTable1.getSelectedRows())[0];
          int startCol=(jTable1.getSelectedColumns())[0];
          try
          {
             String trstring= (String)(system.getContents(this).getTransferData(DataFlavor.stringFlavor));
             String[] st1 = trstring.split("\n"); // excel sometime use \n and sometime use \r\n as delimiter
             // so we should use \n. 
             for(int i=0;i<st1.length;i++)
             {
                rowstring=st1[i];
                // remove the remaining \r when not copied from excel
                if(rowstring.endsWith("\r")){
                	rowstring = rowstring.substring(0,rowstring.length()-1);
                }
                String[] st2 = rowstring.split("\t");
                for(int j=0;j<st2.length;j++)
                {
                   value=(String)st2[j];
                   if (startRow+i< jTable1.getRowCount()  &&
                       startCol+j< jTable1.getColumnCount())
                      jTable1.setValueAt(value,startRow+i,startCol+j);
               }
            }
         }
         catch(Exception ex){ex.printStackTrace();}
      }
   }
}
